#pragma once
#include <iostream>
#include <cstring>
#include <sys/types.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/ipc.h>
#include <sys/shm.h>
#include <fcntl.h>
using namespace std;

#define size 4096 // os开辟sh是4096的整数倍

const char *pathname = "/home/lr/lrr/linux/day10/shm";
const int porj_id = 0x12345678;

#define filename "fifo"

key_t GetKey()
{
    key_t key = ftok(pathname, porj_id);
    if (key < 0)
    {
        cerr << "errno :" << errno << "errnostring : " << strerror(errno) << endl;
        exit(1);
    }
    return key;
}

int CreateShmHelper(key_t key, int falg)
{
    int shmid = shmget(key, size, falg);
    if (shmid < 0)
    {
        cerr << "errno :" << errno << "errnostring : " << strerror(errno) << endl;
        exit(1);
    }
    return shmid;
}

string ToHex(int i)
{
    char buff[1024];
    sprintf(buff, "0x%x", i);
    return buff;
}

void DeleteShm(void *s)
{
    int dt = shmdt(s);
    if (dt < 0)
    {
        cerr << "errno :" << errno << "errnostring : " << strerror(errno) << endl;
        exit(1);
    }
}

void CleanShm(int shmid)
{
    int c = shmctl(shmid, IPC_RMID, nullptr);
    if (c < 0)
    {
        cerr << "errno :" << errno << "errnostring : " << strerror(errno) << endl;
        exit(1);
    }
}

// 对于shmget的falg参数
//  IPC_CREAT   如果没有就创建，如果存在就返回shmid
//  IPC_EXCL    该选项不能单独使用，需要和IPC_CREAT 配合使用 : 如果不创在就创建，如果存在就报错 （这样可以保证shm是第一次被创建的）
//  在falg 中还可以添加权限 这里权限类似于文件权限
int CreateShm(key_t key)
{
    return CreateShmHelper(key, IPC_CREAT | IPC_EXCL | 0664);
}

int GetShm(key_t key)
{
    return CreateShmHelper(key, IPC_CREAT);
}

void Createfifo()
{
    int n = mkfifo(filename, 0666);
    if (n < 0)
    {
        cerr << "errno : " << errno << "errnostring : " << strerror(errno) << endl;
        exit(1);
    }
}
